# coding=utf-8
# Copyright (C) idata team - All Rights Reserved
#
# @Version:   3.10.9
# @Software:  PyCharm
# @FileName:  lm_fc.py
# @CTime:     2023/6/2 14:01   
# @Author:    yhy
# @Email:     yhy@cyber.com
# @UTime:     2023/6/2 14:01
#
# @Description:
#     
#     xxx
#
import logging
from typing import NewType, Any, Optional

logger = logging.getLogger(__name__)


class BERT_FC(nn.Module):
    def __init__(self):
        pass

    def forward(self, x):
        mask = torch.randn(2, 3)
        mask = mask.to(x)



if __name__ == '__main__':
    model = BERT_FC()

